#!/usr/bin/env python3
"""
手动数据收集脚本
"""

import os
import sys
import argparse
from pathlib import Path

# 添加项目根目录到Python路径
project_root = Path(__file__).parent
sys.path.insert(0, str(project_root))

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Didi.settings.dev')

import django
django.setup()

from ai.scripts.scheduled_data_collection import ScheduledDataCollection

def main():
    """主函数"""
    parser = argparse.ArgumentParser(description='手动数据收集工具')
    parser.add_argument('--type', choices=['daily', 'weekly', 'health'], 
                       default='daily', help='收集类型')
    parser.add_argument('--force', action='store_true', help='强制执行')
    
    args = parser.parse_args()
    
    print("=" * 50)
    print("智能客服数据收集工具")
    print("=" * 50)
    
    scheduler = ScheduledDataCollection()
    
    if args.type == 'daily':
        print("执行每日数据收集...")
        success = scheduler.daily_collection()
        if success:
            print("✅ 每日数据收集完成")
        else:
            print("❌ 每日数据收集失败")
            
    elif args.type == 'weekly':
        print("执行每周完整数据收集...")
        success = scheduler.weekly_full_collection()
        if success:
            print("✅ 每周完整数据收集完成")
        else:
            print("❌ 每周完整数据收集失败")
            
    elif args.type == 'health':
        print("执行系统健康检查...")
        success = scheduler.health_check()
        if success:
            print("✅ 系统健康检查完成")
        else:
            print("❌ 系统健康检查失败")

if __name__ == "__main__":
    main() 